import qs from 'qs';
import axios from "axios";
import {isFunction} from "@/utils";

let pendingMap = new Map();
const getPendingUrl = (config) => [config.method, config.url, qs.stringify(config.data), qs.stringify(config.params)].join('&')

class AxiosCancel {
  /**
   * @description 添加请求
   * @param config
   */
  addPending(config) {
    this.removePending(config)
    const url = getPendingUrl(config)
    config.cancelToken = config.cancelToken || new axios.CancelToken((cancel) => {
      if (!pendingMap.has(url)) {
        // 如果pending中不存在当前请求，则添加进去
        pendingMap.set(url, cancel)
      }
    })

  }

  /**
   * @description 移除请求
   * @param config
   */
  removePending(config) {
    const url = getPendingUrl(config);
    if (pendingMap.has(url)) {
      // 如果在pending中存在当前请求标识，需要取消当前请求，并且移除
      const cancel = pendingMap.get(url)
      cancel && cancel();
      pendingMap.delete(url)
    }
  }

  removeAllPending() {
    pendingMap.forEach(cancel => {
      cancel && isFunction(cancel) && cancel()
    })
    pendingMap.clear();
  }

  reset() {
    pendingMap = new Map()
  }
}

export default AxiosCancel;
